TimestampDataType.java

package net.sourceforge.mayfly.datastore.types;

import net.sourceforge.mayfly.MayflyException;
import net.sourceforge.mayfly.datastore.Cell;
import net.sourceforge.mayfly.datastore.StringCell;
import net.sourceforge.mayfly.datastore.TimestampCell;
import net.sourceforge.mayfly.evaluation.Value;
import net.sourceforge.mayfly.parser.Location;

import org.apache.commons.lang.StringEscapeUtils;
import org.joda.time.IllegalFieldValueException;
import org.joda.time.LocalDate;
import org.joda.time.LocalDateTime;

import java.util.regex.Matcher;
import java.util.regex.Pattern;

public class TimestampDataType extends DataType {

    @Override
    public Cell coerce(Value value, String columnName) {
        if (value.value instanceof StringCell) {
            return stringToDate(value.value.asString(), value.location);
        }
        else {
            return genericCoerce(value, columnName, 
                "timestamp", TimestampCell.class);
        }
    }

    TimestampCell stringToDate(String text) {
        return stringToDate(text, Location.UNKNOWN);
    }

    TimestampCell stringToDate(String text, Location location) {
        try {
            LocalDateTime stamp = parseTimestamp(text);
            if (stamp != null) {
                return new TimestampCell(stamp);
            }
            else {
                LocalDate date = DateDataType.parseDate(text);
                if (date != null) {
                    LocalDateTime stampFromDate = new LocalDateTime(
                        date.getYear(), date.getMonthOfYear(), date.getDayOfMonth(),
                        0, 0, 0, 000);
                    return new TimestampCell(stampFromDate);
                }
            }
            throw new MayflyException(
                "'" + StringEscapeUtils.escapeSql(text) + 
                "' is not in format yyyy-mm-dd hh:mm:ss",
                location);
        }
        catch (IllegalFieldValueException e) {
            throw new MayflyException(e.getMessage(), location);
        }
    }

    public static LocalDateTime parseTimestamp(String text) {
        Pattern pattern = Pattern.compile(
            "([0-9]{4})-([0-9]{2})-([0-9]{2}) ([0-9]{2}):([0-9]{2}):([0-9]{2})");
        Matcher matcher = pattern.matcher(text);
        if (matcher.matches()) {
            int year = Integer.parseInt(matcher.group(1));
            int month = Integer.parseInt(matcher.group(2));
            int day = Integer.parseInt(matcher.group(3));
            int hour = Integer.parseInt(matcher.group(4));
            int minute = Integer.parseInt(matcher.group(5));
            int second = Integer.parseInt(matcher.group(6));
            return new LocalDateTime(year, month, day, hour, minute, second, 000);
        }
        else {
            return null;
        }
    }
    
    @Override
    public String dumpName() {
        return "TIMESTAMP";
    }

}